/**
 * Copyright (c) 2022 KCloud-Platform-Alibaba Authors. All Rights Reserved.
 * <p>
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * <p>
 *   http://www.apache.org/licenses/LICENSE-2.0
 * <p>
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.laokou.admin.server.infrastructure.server;
import lombok.SneakyThrows;
import lombok.Data;
import org.laokou.common.core.utils.BigDecimalUtil;
import org.laokou.common.core.utils.DateUtil;
import oshi.SystemInfo;
import oshi.hardware.CentralProcessor;
import oshi.hardware.CentralProcessor.TickType;
import oshi.hardware.GlobalMemory;
import oshi.hardware.HardwareAbstractionLayer;
import oshi.software.os.FileSystem;
import oshi.software.os.OSFileStore;
import oshi.software.os.OperatingSystem;
import oshi.util.Util;
import java.io.Serial;
import java.net.InetAddress;
import java.time.LocalDateTime;
import java.util.*;
import java.io.Serializable;
import java.lang.management.ManagementFactory;
import java.util.LinkedList;
import java.util.Properties;
/**
 * 服务器相关信息
 * @author laokou
 */
@Data
public class Server implements Serializable{

    @Serial
    private static final long serialVersionUID = 886551822597986662L;
    private static final int OSHI_WAIT_SECOND = 1000;

    /**
     * CPU相关信息
     */
    private Cpu cpu = new Cpu();

    /**
     * 內存相关信息
     */
    private Mem mem = new Mem();

    /**
     * JVM相关信息
     */
    private Jvm jvm = new Jvm();

    /**
     * 服务器相关信息
     */
    private Sys sys = new Sys();

    /**
     * 磁盘相关信息
     */
    private List<SysFile> files = new LinkedList<>();

    public Cpu getCpu() {
        return cpu;
    }

    public void setCpu(Cpu cpu) {
        this.cpu = cpu;
    }

    public Mem getMem() {
        return mem;
    }

    public void setMem(Mem mem) {
        this.mem = mem;
    }

    public Jvm getJvm() {
        return jvm;
    }

    public void setJvm(Jvm jvm) {
        this.jvm = jvm;
    }

    public Sys getSys() {
        return sys;
    }

    public void setSys(Sys sys) {
        this.sys = sys;
    }

    public List<SysFile> getFiles() {
        return files;
    }

    public void setFiles(List<SysFile> files) {
        this.files = files;
    }

    public void copyTo() {
        SystemInfo si = new SystemInfo();
        HardwareAbstractionLayer hal = si.getHardware();
        setCpuInfo(hal.getProcessor());
        setMemInfo(hal.getMemory());
        setSysInfo();
        setJvmInfo();
        setSysFiles(si.getOperatingSystem());
    }

    /**
     * 设置CPU信息
     */
    private void setCpuInfo(CentralProcessor processor) {
        // CPU信息
        long[] prevTicks = processor.getSystemCpuLoadTicks();
        Util.sleep(OSHI_WAIT_SECOND);
        long[] ticks = processor.getSystemCpuLoadTicks();
        long nice = ticks[TickType.NICE.getIndex()] - prevTicks[TickType.NICE.getIndex()];
        long irq = ticks[TickType.IRQ.getIndex()] - prevTicks[TickType.IRQ.getIndex()];
        long softirq = ticks[TickType.SOFTIRQ.getIndex()] - prevTicks[TickType.SOFTIRQ.getIndex()];
        long steal = ticks[TickType.STEAL.getIndex()] - prevTicks[TickType.STEAL.getIndex()];
        long cSys = ticks[TickType.SYSTEM.getIndex()] - prevTicks[TickType.SYSTEM.getIndex()];
        long user = ticks[TickType.USER.getIndex()] - prevTicks[TickType.USER.getIndex()];
        long iowait = ticks[TickType.IOWAIT.getIndex()] - prevTicks[TickType.IOWAIT.getIndex()];
        long idle = ticks[TickType.IDLE.getIndex()] - prevTicks[TickType.IDLE.getIndex()];
        long totalCpu = user + nice + cSys + idle + iowait + irq + softirq + steal;
        cpu.setCpuNum(processor.getLogicalProcessorCount());
        cpu.setTotal(totalCpu);
        cpu.setSys(cSys);
        cpu.setUsed(user);
        cpu.setWait(iowait);
        cpu.setFree(idle);
    }

    /**
     * 设置内存信息
     */
    private void setMemInfo(GlobalMemory memory) {
        mem.setTotal(memory.getTotal());
        mem.setUsed(memory.getTotal() - memory.getAvailable());
        mem.setFree(memory.getAvailable());
    }

    /**
     * 设置服务器信息
     */
    @SneakyThrows
    private void setSysInfo() {
        Properties props = System.getProperties();
        sys.setComputerName(InetAddress.getLocalHost().getHostName());
        sys.setComputerIp(InetAddress.getLocalHost().getHostAddress());
        sys.setOsName(props.getProperty("os.name"));
        sys.setOsArch(props.getProperty("os.arch"));
        sys.setUserDir(props.getProperty("user.dir"));
    }

    /**
     * 设置Java虚拟机
     */
    private void setJvmInfo() {
        Properties props = System.getProperties();
        jvm.setTotal(Runtime.getRuntime().totalMemory());
        jvm.setMax(Runtime.getRuntime().maxMemory());
        jvm.setFree(Runtime.getRuntime().freeMemory());
        jvm.setVersion(props.getProperty("java.version"));
        jvm.setHome(props.getProperty("java.home"));
    }

    /**
     * 设置磁盘信息
     */
    private void setSysFiles(OperatingSystem os) {
        FileSystem fileSystem = os.getFileSystem();
        List<OSFileStore> fileStores = fileSystem.getFileStores();
        for (OSFileStore fs : fileStores) {
            long free = fs.getUsableSpace();
            long total = fs.getTotalSpace();
            long used = total - free;
            SysFile sysFile = new SysFile();
            sysFile.setDirName(fs.getMount());
            sysFile.setSysTypeName(fs.getType());
            sysFile.setTypeName(fs.getName());
            sysFile.setTotal(convertFileSize(total));
            sysFile.setFree(convertFileSize(free));
            sysFile.setUsed(convertFileSize(used));
            sysFile.setUsage(BigDecimalUtil.multiply(BigDecimalUtil.divide(used, total, 4), 100));
            files.add(sysFile);
        }
    }

    /**
     * 字节转换
     *
     * @param size 字节大小
     * @return 转换后值
     */
    public String convertFileSize(long size) {
        long kb = 1024;
        long mb = kb * 1024;
        long gb = mb * 1024;
        if (size >= gb) {
            return String.format("%.1f GB", (float) size / gb);
        } else if (size >= mb) {
            float f = (float) size / mb;
            return String.format(f > 100 ? "%.0f MB" : "%.1f MB", f);
        } else if (size >= kb) {
            float f = (float) size / kb;
            return String.format(f > 100 ? "%.0f KB" : "%.1f KB", f);
        } else {
            return String.format("%d B", size);
        }
    }

}

@Data
class Jvm implements Serializable {

    @Serial
    private static final long serialVersionUID = 6661783699792848234L;
    /**
     * 当前JVM占用的内存总数(M)
     */
    private double total;

    /**
     * JVM最大可用内存总数(M)
     */
    private double max;

    /**
     * JVM空闲内存(M)
     */
    private double free;

    /**
     * JDK版本
     */
    private String version;

    /**
     * JDK路径
     */
    private String home;

    public double getTotal() {
        return BigDecimalUtil.divide(total, (1024 * 1024), 2);
    }

    public double getMax() {
        return BigDecimalUtil.divide(max, (1024 * 1024), 2);
    }

    public double getFree() {
        return BigDecimalUtil.divide(free, (1024 * 1024), 2);
    }

    public double getUsed() {
        return BigDecimalUtil.divide(total - free, (1024 * 1024), 2);
    }

    public String getVersion() {
        return version;
    }

    public String getHome() {
        return home;
    }

    public double getUsage() {
        return BigDecimalUtil.multiply(BigDecimalUtil.divide(total - free, total, 4), 100);
    }
    /**
     * 获取JDK名称
     */
    public String getName() {
        return ManagementFactory.getRuntimeMXBean().getVmName();
    }

    /**
     * JDK启动时间
     */
    public String getStartTime() {
        long timestamp = ManagementFactory.getRuntimeMXBean().getStartTime();
        LocalDateTime localDateTime = DateUtil.getDateTimeOfTimestamp(timestamp);
        return DateUtil.format(localDateTime, DateUtil.YYYY_MM_DD_HH_MM_SS);
    }

    public static void main(String[] args) {
        Server server = new Server();
        String startTime = server.getJvm().getStartTime();
        System.out.println(startTime);
    }

    /**
     * JDK运行时间
     */
    public String getRunTime() {
        long timestamp = ManagementFactory.getRuntimeMXBean().getStartTime();
        LocalDateTime start = DateUtil.getDateTimeOfTimestamp(timestamp);
        LocalDateTime end = DateUtil.now();
        long day = DateUtil.getDays(start,end);
        long hour = DateUtil.getHours(start,end);
        long min = DateUtil.getMinutes(start,end);
        return day + "天" + hour + "小时" + min + "分钟";
    }
}


@Data
class Cpu implements Serializable {

    @Serial
    private static final long serialVersionUID = 8621293532430186793L;
    /**
     * 核心数
     */
    private int cpuNum;

    /**
     * CPU总的使用率
     */
    private double total;

    /**
     * CPU系统总数
     */
    private double sys;

    /**
     * CPU用户使用率
     */
    private double used;

    /**
     * CPU当前等待率
     */
    private double wait;

    /**
     * CPU当前空闲率
     */
    private double free;


    public double getTotal() {
        return BigDecimalUtil.round(BigDecimalUtil.multiply(total, 100), 2);
    }

    public double getSys() {
        return BigDecimalUtil.round(BigDecimalUtil.multiply(sys / total, 100), 2);
    }

    public double getUsed() {
        return BigDecimalUtil.round(BigDecimalUtil.multiply(used / total, 100), 2);
    }

    public double getWait() {
        return BigDecimalUtil.round(BigDecimalUtil.multiply(wait / total, 100), 2);
    }

    public double getFree() {
        return BigDecimalUtil.round(BigDecimalUtil.multiply(free / total, 100), 2);
    }
}

@Data
class Mem implements Serializable {


    @Serial
    private static final long serialVersionUID = 4618498208469144168L;
    /**
     * 内存总量
     */
    private double total;

    /**
     * 已用内存
     */
    private double used;

    /**
     * 剩余内存
     */
    private double free;

    public double getTotal() {
        return BigDecimalUtil.divide(total, (1024 * 1024 * 1024), 2);
    }

    public double getUsed() {
        return BigDecimalUtil.divide(used, (1024 * 1024 * 1024), 2);
    }


    public double getFree() {
        return BigDecimalUtil.divide(free, (1024 * 1024 * 1024), 2);
    }

    public double getUsage() {
        return BigDecimalUtil.multiply(BigDecimalUtil.divide(used, total, 4), 100);
    }
}

@Data
class Sys implements Serializable {


    @Serial
    private static final long serialVersionUID = -2249049152299436233L;
    /**
     * 服务器名称
     */
    private String computerName;

    /**
     * 服务器Ip
     */
    private String computerIp;

    /**
     * 项目路径
     */
    private String userDir;

    /**
     * 操作系统
     */
    private String osName;

    /**
     * 系统架构
     */
    private String osArch;
}

@Data
class SysFile implements Serializable {

    @Serial
    private static final long serialVersionUID = 2307419364818519046L;
    /**
     * 盘符路径
     */
    private String dirName;

    /**
     * 盘符类型
     */
    private String sysTypeName;

    /**
     * 文件类型
     */
    private String typeName;

    /**
     * 总大小
     */
    private String total;

    /**
     * 剩余大小
     */
    private String free;

    /**
     * 已经使用量
     */
    private String used;

    /**
     * 资源的使用率
     */
    private double usage;
}
